#!/bin/bash
# 该脚本用于实现 root 用户在 Linux 操作系统下的 Docker 中安装单体架构 Redis

# 临时保存并修改语言环境
LANG_temp=$LANG
LANG=zh_CN.UTF-8

# 还原语言环境
function re_lang () {
    # 将语言环境修改回原来的值
    LANG=$LANG_temp
}

# 还原语言环境，并无异常退出
function re_lang_exit () {
    re_lang
    exit 0
}

function success_text () {
    echo -e -n "\033[32m【SUCCESS】\033[0m"
    echo $1
}

function warn_text () {
    echo -e -n "\033[33m【WARN】\033[0m"
    echo $1
}

function error_text () {
    echo -e -n "\033[31m【ERROR】\033[0m"
    echo $1
}

function error_re_lang_exit () {
    echo -e -n "\033[31m【ERROR】\033[0m"
    echo $1
    re_lang_exit
}

# 如果传入的目录不存在，进行目录的创建
function mkdir_404 () {
    if [ ! -d $1 ]; then
        warn_text "目录 $1 不存在，进行目录的创建..."
        mkdir -p $1 || error_re_lang_exit "目录 $1 创建失败"
        success_text "目录 $1 创建成功"
    else
        success_text "目录 $1 已存在，无需进行创建."
    fi
}

# 检查是否安装了社区版 Docker
is_docker_ce_install=`rpm -q docker-ce | grep '未安装软件包.*'`
if [ ${#is_docker_ce_install} -gt 0 ]; then
    error_text '还未安装社区版 Docker，请先进行安装.'
    re_lang_exit
fi

# 启动 Docker 并开启 Docker 开机自启，确保 Docker 处于运行状态
echo "Docker 启动中..."
systemctl start docker || re_lang_exit
echo "开启 Docker 开机自启中..."
systemctl enable docker || re_lang_exit

# 要安装的软件
software=redis

# 要进行安装的版本
version_default=latest
read -p "请输入要安装的 $software 版本(默认 $version_default)：" version
if [ -z "$version" ]; then
    version=$version_default
fi

# Docker 镜像名称及其版本
image_name=$software:$version

# Docker 安装软件相关映射目录和文件的家目录
home_default=/home/docker/$software/$version
read -p "请输入 Docker 安装 $software 相关映射目录和文件的家目录(默认 $home_default)：" home
if [ -z "$home" ]; then
    home=$home_default
fi

# 需要进行映射的目录
map_dir=( /data /conf /log )

# 目录如果不存在，进行目录的创建
mkdir_404 $home
read -p "是否删除原先存在的映射目录(y-确认删除)：" -n 1 is_rm
echo
for (( i=0; i<${#map_dir[*]}; i++ )); do
    if [ "$is_rm" = "y" ]; then 
        # 删除原先存在的映射目录
        rm -rf $home${map_dir[i]}
	warn_text "已删除 $home${map_dir[i]} 目录及其子目录和其中的文件"
    fi
    mkdir_404 $home${map_dir[i]}
done

# 访问密码
while true; do
    read -p "请输入 $software 服务的访问密码：" redis_pwd
    if [[ -n "${redis_pwd}" ]]; then
        break
    fi
done

# 编写配置文件
# 没有进行注释说明的配置项是我目前没有使用到的配置，从 Redis 默认配置文件中复制而来，
# 在 Redis 默认配置文件中有进行配置
echo "开始编写 $software 配置文件..."
tee $home/conf/$software.conf <<- EOF
################################## NETWORK #####################################
# 指定可以访问当前 Redis 服务的客户端 IP，为了使所有其它客户端都可访问，一般要将其注释掉
# bind 127.0.0.1 -::1
# 关闭保护模式，以确保其它客户端可以连接 Redis，否则只允许本机的客户端访问
protected-mode no
# Redis 服务监听端口号，默认 6379
port 6379
# TCP 连接的队列的长度，其主要用于解决高并发场景下客户端慢连接问题，生产环境下（特别是高并发场景下），backlog 的值最好要大一些
tcp-backlog 511
# 当客户端与 Redis 间的空闲时间超过该时长后，客户端和 Redis 服务器的连接会自动断开。单位秒。默认值为 0，表示永远不超时。
timeout 0
# 设置 Redis 检测与其连接的所有客户端的存活性时间间隔，单位秒，一般是在空闲超时 timeout 设置为 0 时进行配置。
tcp-keepalive 300
################################# GENERAL #####################################
# 配置 Redis 启动不采用守护进程方式，避免与 Docker 启动容器 -d(以守护进程执行) 选项发生冲突
daemonize no
# 指定 Redis 运行时 pid（Redis 进程 id）写入的文件
pidfile /var/run/redis_6379.pid
# 配置日志级别。notice：可以获取到在生产中想获取到的适当多的信息，默认级别。
loglevel notice
# 指定 Redis 运行日志记录的位置
logfile "/var/log/redis/redis-server.log"
# 设置数据库的数量
databases 16
# 只有在交互式会话中才会显示 Redis Logo，否则始终在启动日志中显示 Redis Logo。
always-show-logo no
set-proc-title yes
proc-title-template "{title} {listen-addr} {server-mode}"
locale-collate ""
################################ SNAPSHOTTING  ################################
stop-writes-on-bgsave-error yes
rdbcompression yes
rdbchecksum yes
dbfilename dump.rdb
rdb-del-sync-files no
dir ./
################################# REPLICATION #################################
replica-serve-stale-data yes
replica-read-only yes
repl-diskless-sync yes
repl-diskless-sync-delay 5
repl-diskless-sync-max-replicas 0
repl-diskless-load disabled
repl-disable-tcp-nodelay no
replica-priority 100
################################## SECURITY ###################################
acllog-max-len 128
# 设置客户端访问 Redis 服务时的密码
requirepass $redis_pwd
############################# LAZY FREEING ####################################
lazyfree-lazy-eviction no
lazyfree-lazy-expire no
lazyfree-lazy-server-del no
replica-lazy-flush no
lazyfree-lazy-user-del no
lazyfree-lazy-user-flush no
############################ KERNEL OOM CONTROL ##############################
oom-score-adj no
oom-score-adj-values 0 200 800
disable-thp yes
appendonly no
appendfilename "appendonly.aof"
appenddirname "appendonlydir"
appendfsync everysec
no-appendfsync-on-rewrite no
auto-aof-rewrite-percentage 100
auto-aof-rewrite-min-size 64mb
aof-load-truncated yes
aof-use-rdb-preamble yes
aof-timestamp-enabled no
################################## SLOW LOG ###################################
slowlog-log-slower-than 10000
slowlog-max-len 128
################################ LATENCY MONITOR ##############################
latency-monitor-threshold 0
############################# EVENT NOTIFICATION ##############################
notify-keyspace-events ""
############################### ADVANCED CONFIG ###############################
hash-max-listpack-entries 512
hash-max-listpack-value 64
list-max-listpack-size -2
list-compress-depth 0
set-max-intset-entries 512
set-max-listpack-entries 128
set-max-listpack-value 64
zset-max-listpack-entries 128
zset-max-listpack-value 64
hll-sparse-max-bytes 3000
stream-node-max-bytes 4096
stream-node-max-entries 100
activerehashing yes
client-output-buffer-limit normal 0 0 0
client-output-buffer-limit replica 256mb 64mb 60
client-output-buffer-limit pubsub 32mb 8mb 60
hz 10
dynamic-hz yes
aof-rewrite-incremental-fsync yes
rdb-save-incremental-fsync yes
########################### ACTIVE DEFRAGMENTATION #######################
jemalloc-bg-thread yes
EOF

# 如果配置文件编写失败
if [ $? -ne 0 ]; then
    error_re_lang_exit "编写 $software 配置文件失败"
else
    success_text "编写 $software 配置文件成功"
fi

# 创建日志文件
echo "创建 $software 日志文件..."
touch $home/log/redis-server.log || error_re_lang_exit "创建 $software 日志文件失败"
success_text "创建 $software 日志文件成功"
# 修改日志文件权限，防止容器无法访问和修改日志文件
echo "修改 $software 日志文件权限为 777..."
chmod 777 $home/log/redis-server.log || error_re_lang_exit "修改 $software 日志文件权限为 777 失败"
success_text "修改 $software 日志文件权限为 777 成功."

# 查询 Docker 中是否存在指定镜像，如果不存在，则进行镜像的拉取
image=`docker images -a | grep "$software" | grep "$version"`
if [ -z "$image" ]; then
    warn_text "Docker 中不存在 $image_name 镜像"
    echo "开始拉取镜像 $image_name..."
    docker pull $image_name || re_lang_exit
    success_text "拉取 $image_name 镜像成功"
else
    success_text "镜像 $image_name 已存在，无需进行拉取"
fi

# 容器名称
container_name=$software$version

# 判断容器是否存在，如果存在，手动输入指定容器名称
while true; do
    if [ ! -z "`docker ps -a | grep \"$container_name\"`" ]; then
        error_text "名称为 $container_name 的 Docker 容器已经存在"
        read -p "请输入新 Docker 容器名称：" container_name
    else
	if [ -z "$container_name" ]; then
	    error_text "Docker 容器名称不能为空."
	    read -p "请输入新 Docker 容器名称：" container_name
	    continue
	fi
        break
    fi
done

# 容器运行时的端口，即容器的端口映射
port_default=6379
while true; do
    read -p "请输入容器运行映射的端口号(默认 $port_default)：" port
    if [ -z "$port" ]; then
        port=$port_default
    fi
    # 判断端口是否被占用
    if [ ! -z "`lsof -i:$port`" ]; then
        error_text "端口 $port 已经被占用，请选择未被占用的端口号."
    else
	break
    fi
done

# 运行容器
docker run \
--privileged=true \
--restart=always \
-p $port:6379 \
--name $container_name \
-v $home/conf/redis.conf:/etc/redis/redis.conf \
-v $home/data:/data \
-v $home/log:/var/log/redis \
-d $image_name \
redis-server /etc/redis/redis.conf \
|| error_re_lang_exit "$software 容器启动运行失败"

success_text "$software 容器启动运行成功."
echo "$software 容器运行情况如下："
docker ps -f name=$container_name

# 还原语言环境
re_lang
